# -*- coding: utf-8 -*-

import os

import asr_utils
from asr_config import Config
from asr_model2 import ASR



conf = Config()

# 加载数据
(train_wav_files, train_text_labels), (test_wav_files, test_text_labels) = asr_utils.get_wavs_lables()

# 使用完整数据集构建字典
words_size, words, word_num_map = asr_utils.create_dict(train_text_labels+test_text_labels)

# 构建训练模型
train_model = ASR(train_wav_files, train_text_labels, words_size, words, word_num_map, is_training=True)
train_model.build_train()

